package com.huawei.statistics.utils;

import java.io.*;
import java.util.Properties;

/**
 * 获取属性文件工具类
 */
public class PropertiesUtil {
    private String properiesName = "";
    private static PropertiesUtil propertiesUtil;

    public static PropertiesUtil getInstances() {
        if (propertiesUtil == null)
            propertiesUtil = new PropertiesUtil("sysConfig.properties");
        return propertiesUtil;
    }

    public PropertiesUtil() {

    }

    public PropertiesUtil(String fileName) {
        this.properiesName = fileName;
    }

    /**
     * 通过key获取value值
     *
     * @param key
     * @return
     */
    public String readProperty(String key) {
        String value = "";
        InputStream is = null;
        try {
            is = PropertiesUtil.class.getClassLoader().getResourceAsStream(
                    properiesName);
            Properties p = new Properties();
            p.load(is);
            value = p.getProperty(key);
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            try {
                is.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
        return value;
    }

    /**
     * 获取 Properties 类
     *
     * @return
     */
    public Properties getProperties() {
        Properties p = new Properties();
        InputStream is = null;
        try {
            is = PropertiesUtil.class.getClassLoader().getResourceAsStream(
                    properiesName);
            p.load(is);
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            try {
                is.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
        return p;
    }

    /**
     * 向属性文件写值
     *
     * @param key
     * @param value
     */
    public void writeProperty(String key, String value) {
        InputStream is = null;
        OutputStream os = null;
        Properties p = new Properties();
        try {
            is = new FileInputStream(properiesName);
            p.load(is);
            os = new FileOutputStream(PropertiesUtil.class.getClassLoader().getResource(properiesName).getFile());

            p.setProperty(key, value);
            p.store(os, key);
            os.flush();
            os.close();
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            try {
                if (null != is)
                    is.close();
                if (null != os)
                    os.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }

    }

    private static void main(String[] args) {
        System.out.println("1212121");
        PropertiesUtil p = new PropertiesUtil("sysConfig.properties");
        Properties pro = p.getProperties();
        System.out.println(pro.getProperty("randCodeLength"));
        System.out.println(p.readProperty("randCodeLength"));
    }

}
